package com.sz.szaiagent.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 创建上下文查询增强器的工厂
 * 该类用于创建上下文查询增强器，结合空上下文提示模板，提升查询的语义理解和响应质量。
 */
public class LoveAppContextualQueryAugmenterFactory {

    /**
     * 创建上下文查询增强器实例
     * @return 上下文查询增强器实例
     */
    public static ContextualQueryAugmenter createInstance() {
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉，我只能回答恋爱相关的问题，别的没办法帮到您哦，
                有问题可以联系编程导航客服 https://codefather.cn
                """); // 创建空上下文提示模板
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false) // 设置是否允许空上下文
                .emptyContextPromptTemplate(emptyContextPromptTemplate) // 设置空上下文提示模板
                .build(); // 构建上下文查询增强器
    }
}
